package database

import (
	"fmt"
	"log"
	"os"
	"time"
	
	data "fiber-layout/databases/migrations"
	"fiber-layout/provider/config"
	log2 "fiber-layout/provider/logger"
	"gorm.io/driver/postgres"
	"gorm.io/gorm"
	"gorm.io/gorm/logger"
)

var DB *gorm.DB

func InitPGSql() {
	var err error
	dsn := fmt.Sprintf(
		"host=%s user=%s password=%s dbname=%s port=%d sslmode=disable TimeZone=Asia/Shanghai",
		config.Config.GetString("database.postgres.host"),
		config.Config.GetString("database.postgres.user"),
		config.Config.GetString("database.postgres.password"),
		config.Config.GetString("database.postgres.dbname"),
		config.Config.GetInt("database.postgres.port"),
	)
	
	//调试模式
	IsDebug := logger.Silent
	if config.Config.GetBool("debug") {
		IsDebug = logger.Info
	}
	newLogger := logger.New(
		log.New(os.Stdout, "\r\n", log.LstdFlags), // io writer（日志输出的目标，前缀和日志包含的内容——译者注）
		logger.Config{
			SlowThreshold:             time.Second, // 慢 SQL 阈值
			LogLevel:                  IsDebug,     // 日志级别
			IgnoreRecordNotFoundError: true,        // 忽略ErrRecordNotFound（记录未找到）错误
			Colorful:                  false,       // 禁用彩色打印
		},
	)
	DB, err = gorm.Open(
		postgres.Open(dsn), &gorm.Config{
			// 禁用外键
			DisableForeignKeyConstraintWhenMigrating: true,
			SkipDefaultTransaction: true,
			Logger:                 newLogger,
		},
	)
	
	if err != nil {
		log2.Log.Fatal("链接数据库失败：" + err.Error())
	}
	
	sqlDB, _ := DB.DB()
	//  设置空闲连接池中连接的最大数量
	sqlDB.SetMaxIdleConns(10)
	//  设置打开数据库连接的最大数量。
	sqlDB.SetMaxOpenConns(100)
	//  设置了连接可复用的最大时间。
	sqlDB.SetConnMaxLifetime(time.Hour)
	
	//迁移及表注释
	data.AutoMigrate(DB)
}
